package com.feri.fitness.network.censor.core;

import com.baidu.aip.contentcensor.AipContentCensor;
import com.baidu.aip.contentcensor.EImgType;
import org.json.JSONObject;

/**
 * ━━━━━━神兽出没━━━━━━
 * 　　　┏┓　　　┏┓
 * 　　┏┛┻━━━┛┻┓
 * 　　┃　　　　　　 ┃
 * 　　┃　　　━　　　┃
 * 　　┃　┳┛　┗┳　  ┃
 * 　　┃　　　　　　 ┃
 * 　　┃　　　┻　　　┃
 * 　　┃　　　　　　　┃
 * 　　┗━┓　　　┏━┛Code is far away from bug with the animal protecting
 * 　　　　┃　　　┃    神兽保佑,代码无bug
 * 　　　　┃　　　┃
 * 　　　　┃　　　┗━━━┓
 * 　　　　┃　　　　　　　┣┓
 * 　　　　┃　　　　　　　┏┛
 * 　　　　┗┓┓┏━┳┓┏┛
 * 　　　　　┃┫┫　┃┫┫
 * 　　　　　┗┻┛　┗┻┛
 * ━━━━━━永无BUG!━━━━━━
 *
 * @Description:
 * @Author: 邢朋辉
 * @Date: 2022/12/2 10:33
 */
public class BaiduCensorUtil {
    //设置APPID/AK/SK
    public static final String APP_ID = "17214730";
    public static final String API_KEY = "WnrcZnQNbveI7UG7sDroOm1K";
    public static final String SECRET_KEY = "FUui3jN440l9eLN8B335C3VS62HvtafU";
    private static AipContentCensor client;
    static {
        client=new AipContentCensor(APP_ID, API_KEY, SECRET_KEY);
    }
    /**
     * 审核文本
     * @param msg 需要审核的文本内容
     * @return 是否合规*/
    public static boolean censorTxt(String msg){
        //发起审核
        JSONObject response = client.textCensorUserDefined(msg);
        System.err.println(response);
        //验证审核是否通过
        return response.getInt("conclusionType")==1;
    }
    /**
     * 审核图片
     * @param data 图片内容
     * @return  是否合规*/
    public static boolean censorImg(byte[] data){
        JSONObject response = client.imageCensorUserDefined(data, null);
        System.err.println(response);
        //验证审核是否通过
        return response.getInt("conclusionType")==1;
    }
    /**
     * 审核图片
     * @param path 图片路径
     * @return  是否合规*/
    public static boolean censorImg(String path){
        JSONObject response = client.imageCensorUserDefined(path, EImgType.FILE, null);
        System.err.println(response);
        //验证审核是否通过
        return response.getInt("conclusionType")==1;
    }
}
